if statement in python example